import torch

import io
import os
from parameters import DATA_SOURCE

root = os.path.split(os.path.realpath(__file__))[0] + '\\dict_dataset.pt'
# text_dic = {}
# with io.open(DATA_SOURCE,encoding='utf-8') as f:
#     for index,line in enumerate(f):
#         text_dic[str(index)] = line


# torch.save(text_dic,root)

text_dic = torch.load(root)

for i in range(20):
    print(text_dic[str(i)])